import torch

EMB_MODEL_DICT = {
    "bge-small":"./Embeddings/bge-small-zh-v1.5",
    "bge-base": "/yfzx/models/embeddings/bge-base-zh",
    "bge-large":"./Embeddings/bge-large-zh-v1.5"
    }

LLM_URL = "http://172.16.1.190:7667/job"

EMB_MODEL_TYPE = "bge-base"
EMB_DEVICE = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"

MYSQL_DICT = {
    "host":"172.16.2.162",
    "port":3308,
    "user":"bigmodel",
    "password":"Bigmodel20231103",
    "database":"bigmodeldb",
    "autocommit":True,
    "charset":"utf8mb4",
    "chaconnect_timeoutrset":100,
}

DB_NAME = "T_GD2"
